// TrainingTask.java
package com.ruoyi.insect.domain;

import com.ruoyi.common.annotation.Excel;
import lombok.Data;
import org.apache.commons.lang3.builder.ToStringBuilder;

import java.util.Date;


import org.apache.commons.lang3.builder.ToStringBuilder;
import org.apache.commons.lang3.builder.ToStringStyle;
import com.ruoyi.common.core.domain.BaseEntity;

import java.util.Date;

/**
 * 训练任务对象 training_task
 *
 * @author ruoyi
 * @date 2024-03-20
 */
public class TrainingTask extends BaseEntity
{
    private static final long serialVersionUID = 1L;

    /** 任务ID */
    private String id;

    /** 任务名称 */
    @Excel(name = "任务名称")
    private String taskName;

    /** 任务描述 */
    @Excel(name = "任务描述")
    private String description;

    /** 训练轮数 */
    @Excel(name = "训练轮数")
    private Integer epochs;

    /** 学习率 */
    @Excel(name = "学习率")
    private Double learningRate;

    /** 批大小 */
    @Excel(name = "批大小")
    private Integer batchSize;

    /** 模型类型 */
    @Excel(name = "模型类型")
    private String modelType;

    /** 是否使用预训练权重（0否，1是） */
    @Excel(name = "是否使用预训练权重", readConverterExp = "0=否,1=是")
    private Integer usePretrained;

    /** 任务状态 */
    @Excel(name = "任务状态")
    private String status;

    /** 准确率 */
    @Excel(name = "准确率")
    private Double accuracy;

    /** 开始时间 */
    @Excel(name = "开始时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss")
    private Date startTime;

    /** 结束时间 */
    @Excel(name = "结束时间", width = 30, dateFormat = "yyyy-MM-dd HH:mm:ss")
    private Date endTime;

    /** 备注 */
    @Excel(name = "备注")
    private String remark;

    /** 训练日志 */
    private String logs;

    public void setId(String id)
    {
        this.id = id;
    }

    public String getId()
    {
        return id;
    }
    public void setTaskName(String taskName)
    {
        this.taskName = taskName;
    }

    public String getTaskName()
    {
        return taskName;
    }
    public void setDescription(String description)
    {
        this.description = description;
    }

    public String getDescription()
    {
        return description;
    }
    public void setEpochs(Integer epochs)
    {
        this.epochs = epochs;
    }

    public Integer getEpochs()
    {
        return epochs;
    }
    public void setLearningRate(Double learningRate)
    {
        this.learningRate = learningRate;
    }

    public Double getLearningRate()
    {
        return learningRate;
    }
    public void setBatchSize(Integer batchSize)
    {
        this.batchSize = batchSize;
    }

    public Integer getBatchSize()
    {
        return batchSize;
    }
    public void setModelType(String modelType)
    {
        this.modelType = modelType;
    }

    public String getModelType()
    {
        return modelType;
    }
    public void setUsePretrained(Integer usePretrained)
    {
        this.usePretrained = usePretrained;
    }

    public Integer getUsePretrained()
    {
        return usePretrained;
    }
    public void setStatus(String status)
    {
        this.status = status;
    }

    public String getStatus()
    {
        return status;
    }
    public void setAccuracy(Double accuracy)
    {
        this.accuracy = accuracy;
    }

    public Double getAccuracy()
    {
        return accuracy;
    }
    public void setStartTime(Date startTime)
    {
        this.startTime = startTime;
    }

    public Date getStartTime()
    {
        return startTime;
    }
    public void setEndTime(Date endTime)
    {
        this.endTime = endTime;
    }

    public Date getEndTime()
    {
        return endTime;
    }
    public void setRemark(String remark)
    {
        this.remark = remark;
    }

    public String getRemark()
    {
        return remark;
    }
    public void setLogs(String logs)
    {
        this.logs = logs;
    }

    public String getLogs()
    {
        return logs;
    }

    @Override
    public String toString() {
        return new ToStringBuilder(this,ToStringStyle.MULTI_LINE_STYLE)
                .append("id", getId())
                .append("taskName", getTaskName())
                .append("description", getDescription())
                .append("epochs", getEpochs())
                .append("learningRate", getLearningRate())
                .append("batchSize", getBatchSize())
                .append("modelType", getModelType())
                .append("usePretrained", getUsePretrained())
                .append("status", getStatus())
                .append("accuracy", getAccuracy())
                .append("startTime", getStartTime())
                .append("endTime", getEndTime())
                .append("remark", getRemark())
                .append("createBy", getCreateBy())
                .append("createTime", getCreateTime())
                .append("logs", getLogs())
                .toString();
    }
}
